package com.tianji.remark.task;

import cn.hutool.core.collection.CollUtil;
import com.tianji.common.autoconfigure.mq.RabbitMqHelper;
import com.tianji.common.constants.CacheConstant;
import com.tianji.common.constants.MqConstants;
import com.tianji.common.domain.dto.LikedTimesDTO;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.ZSetOperations;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

@Slf4j
@Component
public class LikeRecordSyncToMysqlTask {

    @Autowired
    private StringRedisTemplate redisTemplate;
    @Autowired
    private RabbitMqHelper mqHelper;

    // 秒 分 时 天 月 周 年
    // 每隔10秒钟，把redis中的数据读取出来，发送mq消息
    @Scheduled(cron = "*/10 * * * * ?")
    public void likeRecordSyncToMysql(){
        log.info("点赞数量同步到数据据任务开始。。。");
        int pageNo = 1;
        while(true){
            // 分页，每次取100条数据
            Set<ZSetOperations.TypedTuple<String>> typedTuples = redisTemplate.opsForZSet().popMax(CacheConstant.LIKE_COUNT, 100);
            if(CollUtil.isEmpty(typedTuples)){
                log.info("没有更多需要同步的数据");
                return;
            }
            List<LikedTimesDTO> msgList = new ArrayList<>();
            for (ZSetOperations.TypedTuple<String> typedTuple : typedTuples) {
                // 点赞的次数
                Double score = typedTuple.getScore();
                //bizType+":"+bizId
                String value = typedTuple.getValue();
                String[] arr = value.split(":");
                String bizType = arr[0];
                String bizId = arr[1];
                LikedTimesDTO msg = new LikedTimesDTO(Long.parseLong(bizId), score.intValue(), bizType);
                msgList.add(msg);
            }
            // 按照bizType分组，一个bizType一条消息
            Map<String, List<LikedTimesDTO>> msgMap = msgList.stream().collect(Collectors.groupingBy(LikedTimesDTO::getBizType));
            // 根据不同的bizType把消息发到不同的队列中
            for (Map.Entry<String, List<LikedTimesDTO>> entry : msgMap.entrySet()) {
                String bizType = entry.getKey();
                List<LikedTimesDTO> msgs = entry.getValue();
                String key = MqConstants.Key.LIKED_TIMES_KEY_TEMPLATE.replace("{}", bizType);
                mqHelper.send(MqConstants.Exchange.LIKE_RECORD_EXCHANGE, key, msgs);
                log.info("发送第{}页点赞数量的消息，bizType：{}， msgList：{}", pageNo++, bizType, msgs);
            }
        }
    }

}
